//
//  FQDArticleFeedCellItem.m
//  FakeQDaily
//
//  Created by mewxiv on 10/7/16.
//  Copyright © 2016 Xensius. All rights reserved.
//

#import "FQDArticleFeedCellItem.h"
#import "FQDRegularArticleFeedCell.h"
#import "FQDLargeArticleFeedCell.h"
#import "FQDFeed.h"


@implementation FQDArticleFeedCellItem

- (void)setRestModel:(id)restModel
{
    [super setRestModel:restModel];
    self.feed = restModel;
}

- (Class)cellClass
{
    if (self.feed.type == FQDFeedTypeLargeArticle) {
        return [FQDLargeArticleFeedCell class];
    } else if (self.feed.type == FQDFeedTypeRegularArticle) {
        return [FQDRegularArticleFeedCell class];
    }
    return [FQDRegularArticleFeedCell class];
}

@end
